class Config:
    # ROOT_PATH = r'./data'
    ROOT_PATH = r'./flower_photos'
    BATCH_SIZE = 16
    MODEL_WEIGHT_PATH = r'./resnet50.pth'
    CLASSES_NUM = 5
    CLASSES = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
